import json

template = {
    'task_type': 'cm',  # Replace 'cm' with 'pgm' for PGM tasks
    'file_type': 'csv',
    'file_path': '/home/yulin/Downloads/full_data/cm_africa.csv',  # Path of the input csv file
    'dv_file_path': '/home/yulin/Downloads/dv_data/dv_cm_africa.csv',  # Path of the dv file
    # 'filter_file_path' and 'filter_mode' are for Hurdle model only. -- uncomment when required
    # 'filter_file_path': '/home/yulin/Downloads/grid_data/grid_conflict_count.csv',  # Path of the grid_conflict file
    # 'filter_mode': 3,  # Filtering mode, 1, 2, 3 denotes Hurdle 1, 2, 3, respectively.
    'engine': 'pyarrow',
    'shift': 6,  # s, which controls the lag of the target dataset, should be consistent with dv file

    # Feature collection for PGM_DV_ONLY
    'features': ['country_name',
                 'decay_12_time_since_ged_dummy_sb',
                 'ged_best_sb',
                 'ged_count_sb',
                 'ged_dummy_sb',
                 'greq_100_ged_best_sb',
                 'greq_1_ged_best_sb',
                 'greq_25_ged_best_sb',
                 'greq_25_splag_1_1_ged_best_sb',
                 'greq_500_ged_best_sb',
                 'greq_500_splag_1_1_ged_best_sb',
                 'greq_5_ged_best_sb',
                 'ln_ged_best_sb',
                 'month',
                 'splag_1_1_ged_best_sb',
                 'splag_1_1_ged_dummy_sb',
                 'stdist_k1_t001_ged_dummy_sb',
                 'stdist_k1_t10_ged_dummy_sb',
                 'stdist_k1_t1_ged_dummy_sb',
                 'time_since_ged_dummy_sb',
                 'time_since_greq_25_ged_best_sb',
                 'time_since_greq_25_splag_1_1_ged_best_sb',
                 'time_since_greq_500_ged_best_sb',
                 'time_since_greq_500_splag_1_1_ged_best_sb',
                 'tlag_10_ged_dummy_sb',
                 'tlag_11_ged_dummy_sb',
                 'tlag_12_ged_dummy_sb',
                 'tlag_1_ged_best_sb',
                 'tlag_1_ged_dummy_sb',
                 'tlag_1_splag_1_1_ged_dummy_sb',
                 'tlag_2_ged_dummy_sb',
                 'tlag_2_splag_1_1_ged_dummy_sb',
                 'tlag_3_ged_dummy_sb',
                 'tlag_3_splag_1_1_ged_dummy_sb',
                 'tlag_4_ged_dummy_sb',
                 'tlag_5_ged_dummy_sb',
                 'tlag_6_ged_dummy_sb',
                 'tlag_7_ged_dummy_sb',
                 'tlag_8_ged_dummy_sb',
                 'tlag_9_ged_dummy_sb',
                 ],
    # Feature collection of CM_BENCHMARK
    # 'features': [
    #     'cdum_1',
    #     'cdum_10',
    #     'cdum_100',
    #     'cdum_101',
    #     'cdum_102',
    #     'cdum_103',
    #     'cdum_104',
    #     'cdum_105',
    #     'cdum_106',
    #     'cdum_107',
    #     'cdum_108',
    #     'cdum_109',
    #     'cdum_11',
    #     'cdum_110',
    #     'cdum_111',
    #     'cdum_112',
    #     'cdum_113',
    #     'cdum_114',
    #     'cdum_115',
    #     'cdum_116',
    #     'cdum_117',
    #     'cdum_118',
    #     'cdum_119',
    #     'cdum_12',
    #     'cdum_120',
    #     'cdum_121',
    #     'cdum_122',
    #     'cdum_123',
    #     'cdum_124',
    #     'cdum_125',
    #     'cdum_126',
    #     'cdum_127',
    #     'cdum_128',
    #     'cdum_129',
    #     'cdum_13',
    #     'cdum_130',
    #     'cdum_131',
    #     'cdum_132',
    #     'cdum_133',
    #     'cdum_134',
    #     'cdum_135',
    #     'cdum_136',
    #     'cdum_137',
    #     'cdum_138',
    #     'cdum_139',
    #     'cdum_14',
    #     'cdum_140',
    #     'cdum_141',
    #     'cdum_142',
    #     'cdum_143',
    #     'cdum_144',
    #     'cdum_145',
    #     'cdum_146',
    #     'cdum_147',
    #     'cdum_148',
    #     'cdum_149',
    #     'cdum_15',
    #     'cdum_150',
    #     'cdum_151',
    #     'cdum_152',
    #     'cdum_153',
    #     'cdum_154',
    #     'cdum_155',
    #     'cdum_156',
    #     'cdum_157',
    #     'cdum_158',
    #     'cdum_159',
    #     'cdum_16',
    #     'cdum_160',
    #     'cdum_161',
    #     'cdum_162',
    #     'cdum_163',
    #     'cdum_164',
    #     'cdum_165',
    #     'cdum_166',
    #     'cdum_167',
    #     'cdum_168',
    #     'cdum_169',
    #     'cdum_17',
    #     'cdum_170',
    #     'cdum_171',
    #     'cdum_172',
    #     'cdum_173',
    #     'cdum_174',
    #     'cdum_175',
    #     'cdum_176',
    #     'cdum_177',
    #     'cdum_178',
    #     'cdum_179',
    #     'cdum_18',
    #     'cdum_180',
    #     'cdum_181',
    #     'cdum_182',
    #     'cdum_183',
    #     'cdum_184',
    #     'cdum_185',
    #     'cdum_186',
    #     'cdum_187',
    #     'cdum_188',
    #     'cdum_189',
    #     'cdum_19',
    #     'cdum_190',
    #     'cdum_191',
    #     'cdum_192',
    #     'cdum_193',
    #     'cdum_194',
    #     'cdum_195',
    #     'cdum_196',
    #     'cdum_197',
    #     'cdum_198',
    #     'cdum_199',
    #     'cdum_2',
    #     'cdum_20',
    #     'cdum_200',
    #     'cdum_201',
    #     'cdum_202',
    #     'cdum_203',
    #     'cdum_204',
    #     'cdum_205',
    #     'cdum_206',
    #     'cdum_207',
    #     'cdum_208',
    #     'cdum_209',
    #     'cdum_21',
    #     'cdum_210',
    #     'cdum_211',
    #     'cdum_212',
    #     'cdum_213',
    #     'cdum_214',
    #     'cdum_215',
    #     'cdum_216',
    #     'cdum_217',
    #     'cdum_218',
    #     'cdum_219',
    #     'cdum_22',
    #     'cdum_220',
    #     'cdum_221',
    #     'cdum_222',
    #     'cdum_223',
    #     'cdum_224',
    #     'cdum_225',
    #     'cdum_226',
    #     'cdum_227',
    #     'cdum_228',
    #     'cdum_229',
    #     'cdum_23',
    #     'cdum_230',
    #     'cdum_231',
    #     'cdum_232',
    #     'cdum_233',
    #     'cdum_234',
    #     'cdum_235',
    #     'cdum_236',
    #     'cdum_237',
    #     'cdum_238',
    #     'cdum_239',
    #     'cdum_24',
    #     'cdum_240',
    #     'cdum_241',
    #     'cdum_242',
    #     'cdum_243',
    #     'cdum_244',
    #     'cdum_245',
    #     'cdum_246',
    #     'cdum_247',
    #     'cdum_248',
    #     'cdum_249',
    #     'cdum_25',
    #     'cdum_250',
    #     'cdum_251',
    #     'cdum_252',
    #     'cdum_253',
    #     'cdum_254',
    #     'cdum_255',
    #     'cdum_26',
    #     'cdum_27',
    #     'cdum_28',
    #     'cdum_29',
    #     'cdum_3',
    #     'cdum_30',
    #     'cdum_31',
    #     'cdum_32',
    #     'cdum_33',
    #     'cdum_34',
    #     'cdum_35',
    #     'cdum_36',
    #     'cdum_37',
    #     'cdum_38',
    #     'cdum_39',
    #     'cdum_4',
    #     'cdum_40',
    #     'cdum_41',
    #     'cdum_42',
    #     'cdum_43',
    #     'cdum_44',
    #     'cdum_45',
    #     'cdum_46',
    #     'cdum_47',
    #     'cdum_48',
    #     'cdum_49',
    #     'cdum_5',
    #     'cdum_50',
    #     'cdum_51',
    #     'cdum_52',
    #     'cdum_53',
    #     'cdum_54',
    #     'cdum_55',
    #     'cdum_56',
    #     'cdum_57',
    #     'cdum_58',
    #     'cdum_59',
    #     'cdum_6',
    #     'cdum_60',
    #     'cdum_61',
    #     'cdum_62',
    #     'cdum_63',
    #     'cdum_64',
    #     'cdum_65',
    #     'cdum_66',
    #     'cdum_67',
    #     'cdum_68',
    #     'cdum_69',
    #     'cdum_7',
    #     'cdum_70',
    #     'cdum_71',
    #     'cdum_72',
    #     'cdum_73',
    #     'cdum_74',
    #     'cdum_75',
    #     'cdum_76',
    #     'cdum_77',
    #     'cdum_78',
    #     'cdum_79',
    #     'cdum_8',
    #     'cdum_80',
    #     'cdum_81',
    #     'cdum_82',
    #     'cdum_83',
    #     'cdum_84',
    #     'cdum_85',
    #     'cdum_86',
    #     'cdum_87',
    #     'cdum_88',
    #     'cdum_89',
    #     'cdum_9',
    #     'cdum_90',
    #     'cdum_91',
    #     'cdum_92',
    #     'cdum_93',
    #     'cdum_94',
    #     'cdum_95',
    #     'cdum_96',
    #     'cdum_97',
    #     'cdum_98',
    #     'cdum_99',
    #     'fvp_demo',
    #     'fvp_grgdpcap_nonoilrent',
    #     'fvp_grgdpcap_oilrent',
    #     'fvp_grpop200',
    #     'fvp_indepyear',
    #     'fvp_lngdp200',
    #     'fvp_lngdpcap_nonoilrent',
    #     'fvp_lngdpcap_oilrent',
    #     'fvp_lngdppercapita200',
    #     'fvp_population200',
    #     'fvp_prop_discriminated',
    #     'fvp_prop_dominant',
    #     'fvp_prop_excluded',
    #     'fvp_prop_irrelevant',
    #     'fvp_prop_powerless',
    #     'fvp_semi',
    #     'fvp_ssp2_edu_sec_15_24_prop',
    #     'fvp_ssp2_urban_share_iiasa',
    #     'fvp_timeindep',
    #     'fvp_timesincepreindepwar',
    #     'fvp_timesinceregimechange',
    #     'icgcw_alerts',
    #     'icgcw_deteriorated',
    #     'icgcw_improved',
    #     'icgcw_opportunities',
    #     'icgcw_unobserved',
    #     'in_africa',
    #     'reign_age',
    #     'reign_anticipation',
    #     'reign_change_recent',
    #     'reign_couprisk',
    #     'reign_defeat_recent',
    #     'reign_delayed',
    #     'reign_direct_recent',
    #     'reign_elected',
    #     'reign_election_now',
    #     'reign_election_recent',
    #     'reign_exec_ant',
    #     'reign_exec_recent',
    #     'reign_gov_dominant_party',
    #     'reign_gov_foreign_occupied',
    #     'reign_gov_indirect_military',
    #     'reign_gov_military',
    #     'reign_gov_military_personal',
    #     'reign_gov_monarchy',
    #     'reign_gov_oligarchy',
    #     'reign_gov_parliamentary_democracy',
    #     'reign_gov_party_military',
    #     'reign_gov_party_personal',
    #     'reign_gov_party_personal_military_hybrid',
    #     'reign_gov_personal_dictatorship',
    #     'reign_gov_presidential_democracy',
    #     'reign_gov_provisional_civilian',
    #     'reign_gov_provisional_military',
    #     'reign_gov_warlordism',
    #     'reign_indirect_recent',
    #     'reign_irreg_lead_ant',
    #     'reign_irregular',
    #     'reign_lastelection',
    #     'reign_lead_recent',
    #     'reign_leg_ant',
    #     'reign_leg_recent',
    #     'reign_loss',
    #     'reign_male',
    #     'reign_militarycareer',
    #     'reign_nochange_recent',
    #     'reign_pctile_risk',
    #     'reign_precip',
    #     'reign_pt_attempt',
    #     'reign_pt_suc',
    #     'reign_ref_ant',
    #     'reign_ref_recent',
    #     'reign_tenure_months',
    #     'reign_victory_recent',
    #     'splag_1_1_acled_count_ns',
    #     'splag_1_1_acled_count_os',
    #     'splag_1_1_acled_count_pr',
    #     'splag_1_1_acled_count_sb',
    #     'splag_1_1_ged_best_ns',
    #     'splag_1_1_ged_best_os',
    #     'splag_1_1_ged_best_sb',
    #     'time_since_acled_dummy_ns',
    #     'time_since_acled_dummy_os',
    #     'time_since_acled_dummy_pr',
    #     'time_since_acled_dummy_sb',
    #     'time_since_ged_dummy_ns',
    #     'time_since_ged_dummy_os',
    #     'time_since_ged_dummy_sb',
    #     'time_since_greq_100_splag_1_1_ged_best_sb',
    #     'time_since_greq_25_ged_best_ns',
    #     'time_since_greq_25_ged_best_os',
    #     'time_since_greq_25_ged_best_sb',
    #     'time_since_greq_500_ged_best_ns',
    #     'time_since_greq_500_ged_best_os',
    #     'time_since_greq_500_ged_best_sb',
    #     'time_since_splag_1_1_acled_dummy_ns',
    #     'time_since_splag_1_1_acled_dummy_os',
    #     'time_since_splag_1_1_acled_dummy_pr',
    #     'time_since_splag_1_1_acled_dummy_sb',
    #     'time_since_splag_1_1_ged_dummy_ns',
    #     'time_since_splag_1_1_ged_dummy_os',
    #     'time_since_splag_1_1_ged_dummy_sb',
    #     'tlag_12_vdem_v2x_accountability',
    #     'tlag_12_vdem_v2x_api',
    #     'tlag_12_vdem_v2x_civlib',
    #     'tlag_12_vdem_v2x_clphy',
    #     'tlag_12_vdem_v2x_clpol',
    #     'tlag_12_vdem_v2x_clpriv',
    #     'tlag_12_vdem_v2x_corr',
    #     'tlag_12_vdem_v2x_cspart',
    #     'tlag_12_vdem_v2x_delibdem',
    #     'tlag_12_vdem_v2x_diagacc',
    #     'tlag_12_vdem_v2x_divparctrl',
    #     'tlag_12_vdem_v2x_edcomp_thick',
    #     'tlag_12_vdem_v2x_egal',
    #     'tlag_12_vdem_v2x_egaldem',
    #     'tlag_12_vdem_v2x_elecoff',
    #     'tlag_12_vdem_v2x_elecreg',
    #     'tlag_12_vdem_v2x_ex_confidence',
    #     'tlag_12_vdem_v2x_ex_direlect',
    #     'tlag_12_vdem_v2x_ex_hereditary',
    #     'tlag_12_vdem_v2x_ex_military',
    #     'tlag_12_vdem_v2x_ex_party',
    #     'tlag_12_vdem_v2x_execorr',
    #     'tlag_12_vdem_v2x_feduni',
    #     'tlag_12_vdem_v2x_frassoc_thick',
    #     'tlag_12_vdem_v2x_freexp',
    #     'tlag_12_vdem_v2x_freexp_altinf',
    #     'tlag_12_vdem_v2x_gencl',
    #     'tlag_12_vdem_v2x_gencs',
    #     'tlag_12_vdem_v2x_gender',
    #     'tlag_12_vdem_v2x_genpp',
    #     'tlag_12_vdem_v2x_horacc',
    #     'tlag_12_vdem_v2x_hosabort',
    #     'tlag_12_vdem_v2x_hosinter',
    #     'tlag_12_vdem_v2x_jucon',
    #     'tlag_12_vdem_v2x_legabort',
    #     'tlag_12_vdem_v2x_libdem',
    #     'tlag_12_vdem_v2x_liberal',
    #     'tlag_12_vdem_v2x_mpi',
    #     'tlag_12_vdem_v2x_neopat',
    #     'tlag_12_vdem_v2x_partip',
    #     'tlag_12_vdem_v2x_partipdem',
    #     'tlag_12_vdem_v2x_polyarchy',
    #     'tlag_12_vdem_v2x_pubcorr',
    #     'tlag_12_vdem_v2x_regime',
    #     'tlag_12_vdem_v2x_regime_amb',
    #     'tlag_12_vdem_v2x_rule',
    #     'tlag_12_vdem_v2x_suffr',
    #     'tlag_12_vdem_v2x_veracc',
    #     'tlag_12_vdem_v2xcl_acjst',
    #     'tlag_12_vdem_v2xcl_disc',
    #     'tlag_12_vdem_v2xcl_dmove',
    #     'tlag_12_vdem_v2xcl_prpty',
    #     'tlag_12_vdem_v2xcl_rol',
    #     'tlag_12_vdem_v2xcl_slave',
    #     'tlag_12_vdem_v2xcs_ccsi',
    #     'tlag_12_vdem_v2xdd_cic',
    #     'tlag_12_vdem_v2xdd_dd',
    #     'tlag_12_vdem_v2xdd_i_or',
    #     'tlag_12_vdem_v2xdd_i_pi',
    #     'tlag_12_vdem_v2xdd_i_pl',
    #     'tlag_12_vdem_v2xdd_i_rf',
    #     'tlag_12_vdem_v2xdd_toc',
    #     'tlag_12_vdem_v2xdl_delib',
    #     'tlag_12_vdem_v2xeg_eqaccess',
    #     'tlag_12_vdem_v2xeg_eqdr',
    #     'tlag_12_vdem_v2xeg_eqprotec',
    #     'tlag_12_vdem_v2xel_elecparl',
    #     'tlag_12_vdem_v2xel_elecpres',
    #     'tlag_12_vdem_v2xel_frefair',
    #     'tlag_12_vdem_v2xel_locelec',
    #     'tlag_12_vdem_v2xel_regelec',
    #     'tlag_12_vdem_v2xex_elecleg',
    #     'tlag_12_vdem_v2xex_elecreg',
    #     'tlag_12_vdem_v2xlg_elecreg',
    #     'tlag_12_vdem_v2xlg_legcon',
    #     'tlag_12_vdem_v2xlg_leginter',
    #     'tlag_12_vdem_v2xme_altinf',
    #     'tlag_12_vdem_v2xnp_client',
    #     'tlag_12_vdem_v2xnp_pres',
    #     'tlag_12_vdem_v2xnp_regcorr',
    #     'tlag_12_vdem_v2xpe_exlecon',
    #     'tlag_12_vdem_v2xpe_exlgender',
    #     'tlag_12_vdem_v2xpe_exlgeo',
    #     'tlag_12_vdem_v2xpe_exlpol',
    #     'tlag_12_vdem_v2xpe_exlsocgr',
    #     'tlag_12_vdem_v2xps_party',
    #     'tlag_1_greq_1_ged_best_ns',
    #     'tlag_1_greq_1_ged_best_os',
    #     'tlag_1_greq_1_ged_best_sb',
    #     'tlag_1_splag_1_1_ged_best_ns',
    #     'tlag_1_splag_1_1_ged_best_os',
    #     'tlag_1_splag_1_1_ged_best_sb',
    #     'tlag_2_greq_1_ged_best_sb',
    #     'tlag_3_greq_1_ged_best_sb',
    #     'vdem_v2x_accountability',
    #     'vdem_v2x_api',
    #     'vdem_v2x_civlib',
    #     'vdem_v2x_clphy',
    #     'vdem_v2x_clpol',
    #     'vdem_v2x_clpriv',
    #     'vdem_v2x_corr',
    #     'vdem_v2x_cspart',
    #     'vdem_v2x_delibdem',
    #     'vdem_v2x_diagacc',
    #     'vdem_v2x_divparctrl',
    #     'vdem_v2x_edcomp_thick',
    #     'vdem_v2x_egal',
    #     'vdem_v2x_egaldem',
    #     'vdem_v2x_elecoff',
    #     'vdem_v2x_elecreg',
    #     'vdem_v2x_ex_confidence',
    #     'vdem_v2x_ex_direlect',
    #     'vdem_v2x_ex_hereditary',
    #     'vdem_v2x_ex_military',
    #     'vdem_v2x_ex_party',
    #     'vdem_v2x_execorr',
    #     'vdem_v2x_feduni',
    #     'vdem_v2x_frassoc_thick',
    #     'vdem_v2x_freexp',
    #     'vdem_v2x_freexp_altinf',
    #     'vdem_v2x_gencl',
    #     'vdem_v2x_gencs',
    #     'vdem_v2x_gender',
    #     'vdem_v2x_genpp',
    #     'vdem_v2x_horacc',
    #     'vdem_v2x_hosabort',
    #     'vdem_v2x_hosinter',
    #     'vdem_v2x_jucon',
    #     'vdem_v2x_legabort',
    #     'vdem_v2x_libdem',
    #     'vdem_v2x_liberal',
    #     'vdem_v2x_mpi',
    #     'vdem_v2x_neopat',
    #     'vdem_v2x_partip',
    #     'vdem_v2x_partipdem',
    #     'vdem_v2x_polyarchy',
    #     'vdem_v2x_pubcorr',
    #     'vdem_v2x_regime',
    #     'vdem_v2x_regime_amb',
    #     'vdem_v2x_rule',
    #     'vdem_v2x_suffr',
    #     'vdem_v2x_veracc',
    #     'vdem_v2xcl_acjst',
    #     'vdem_v2xcl_disc',
    #     'vdem_v2xcl_dmove',
    #     'vdem_v2xcl_prpty',
    #     'vdem_v2xcl_rol',
    #     'vdem_v2xcl_slave',
    #     'vdem_v2xcs_ccsi',
    #     'vdem_v2xdd_cic',
    #     'vdem_v2xdd_dd',
    #     'vdem_v2xdd_i_or',
    #     'vdem_v2xdd_i_pi',
    #     'vdem_v2xdd_i_pl',
    #     'vdem_v2xdd_i_rf',
    #     'vdem_v2xdd_toc',
    #     'vdem_v2xdl_delib',
    #     'vdem_v2xeg_eqaccess',
    #     'vdem_v2xeg_eqdr',
    #     'vdem_v2xeg_eqprotec',
    #     'vdem_v2xel_elecparl',
    #     'vdem_v2xel_elecpres',
    #     'vdem_v2xel_frefair',
    #     'vdem_v2xel_locelec',
    #     'vdem_v2xel_regelec',
    #     'vdem_v2xex_elecleg',
    #     'vdem_v2xex_elecreg',
    #     'vdem_v2xlg_elecreg',
    #     'vdem_v2xlg_legcon',
    #     'vdem_v2xlg_leginter',
    #     'vdem_v2xme_altinf',
    #     'vdem_v2xnp_client',
    #     'vdem_v2xnp_pres',
    #     'vdem_v2xnp_regcorr',
    #     'vdem_v2xpe_exlecon',
    #     'vdem_v2xpe_exlgender',
    #     'vdem_v2xpe_exlgeo',
    #     'vdem_v2xpe_exlpol',
    #     'vdem_v2xpe_exlsocgr',
    #     'vdem_v2xps_party',
    #     'wdi_ag_lnd_agri_zs',
    #     'wdi_ag_lnd_arbl_zs',
    #     'wdi_ag_lnd_frst_k2',
    #     'wdi_ag_lnd_prcp_mm',
    #     'wdi_ag_lnd_totl_k2',
    #     'wdi_ag_lnd_totl_ru_k2',
    #     'wdi_ag_prd_crop_xd',
    #     'wdi_ag_prd_food_xd',
    #     'wdi_ag_prd_lvsk_xd',
    #     'wdi_ag_srf_totl_k2',
    #     'wdi_ag_yld_crel_kg',
    #     'wdi_bg_gsr_nfsv_gd_zs',
    #     'wdi_bm_klt_dinv_wd_gd_zs',
    #     'wdi_bn_cab_xoka_gd_zs',
    #     'wdi_bx_gsr_ccis_zs',
    #     'wdi_bx_gsr_cmcp_zs',
    #     'wdi_bx_gsr_insf_zs',
    #     'wdi_bx_gsr_mrch_cd',
    #     'wdi_bx_gsr_tran_zs',
    #     'wdi_bx_gsr_trvl_zs',
    #     'wdi_bx_klt_dinv_cd_wd',
    #     'wdi_bx_klt_dinv_wd_gd_zs',
    #     'wdi_bx_trf_pwkr_dt_gd_zs',
    #     'wdi_dt_dod_dect_gn_zs',
    #     'wdi_dt_dod_pvlx_gn_zs',
    #     'wdi_dt_oda_oatl_kd',
    #     'wdi_dt_oda_odat_gn_zs',
    #     'wdi_dt_oda_odat_pc_zs',
    #     'wdi_dt_tds_dect_gn_zs',
    #     'wdi_eg_elc_accs_zs',
    #     'wdi_eg_use_elec_kh_pc',
    #     'wdi_eg_use_pcap_kg_oe',
    #     'wdi_en_pop_slum_ur_zs',
    #     'wdi_en_urb_mcty_tl_zs',
    #     'wdi_ep_pmp_desl_cd',
    #     'wdi_ep_pmp_sgas_cd',
    #     'wdi_fp_cpi_totl',
    #     'wdi_fr_inr_dpst',
    #     'wdi_fr_inr_lndp',
    #     'wdi_gc_dod_totl_gd_zs',
    #     'wdi_ic_bus_ease_xq',
    #     'wdi_iq_cpa_econ_xq',
    #     'wdi_iq_cpa_fisp_xq',
    #     'wdi_iq_cpa_gndr_xq',
    #     'wdi_iq_cpa_macr_xq',
    #     'wdi_iq_cpa_prop_xq',
    #     'wdi_iq_cpa_pubs_xq',
    #     'wdi_iq_cpa_soci_xq',
    #     'wdi_iq_cpa_trad_xq',
    #     'wdi_iq_cpa_tran_xq',
    #     'wdi_ne_con_prvt_pc_kd_zg',
    #     'wdi_ne_dab_totl_kd',
    #     'wdi_ne_dab_totl_zs',
    #     'wdi_ne_exp_gnfs_zs',
    #     'wdi_ne_gdi_totl_zs',
    #     'wdi_ne_imp_gnfs_kd',
    #     'wdi_ne_imp_gnfs_kd_zg',
    #     'wdi_ne_imp_gnfs_zs',
    #     'wdi_ne_rsb_gnfs_zs',
    #     'wdi_ne_trd_gnfs_zs',
    #     'wdi_nv_agr_empl_kd',
    #     'wdi_nv_agr_totl_cd',
    #     'wdi_nv_agr_totl_cn',
    #     'wdi_nv_agr_totl_kd',
    #     'wdi_nv_agr_totl_kd_zg',
    #     'wdi_nv_agr_totl_kn',
    #     'wdi_nv_agr_totl_zs',
    #     'wdi_nv_ind_empl_kd',
    #     'wdi_nv_ind_manf_cd',
    #     'wdi_nv_ind_manf_cn',
    #     'wdi_nv_ind_manf_kd',
    #     'wdi_nv_ind_manf_kd_zg',
    #     'wdi_nv_ind_manf_kn',
    #     'wdi_nv_ind_manf_zs',
    #     'wdi_nv_ind_totl_cd',
    #     'wdi_nv_ind_totl_cn',
    #     'wdi_nv_ind_totl_kd',
    #     'wdi_nv_ind_totl_kd_zg',
    #     'wdi_nv_ind_totl_kn',
    #     'wdi_nv_ind_totl_zs',
    #     'wdi_nv_mnf_chem_zs_un',
    #     'wdi_nv_mnf_fbto_zs_un',
    #     'wdi_nv_mnf_mtrn_zs_un',
    #     'wdi_nv_mnf_othr_zs_un',
    #     'wdi_nv_mnf_tech_zs_un',
    #     'wdi_nv_mnf_txtl_zs_un',
    #     'wdi_nv_srv_empl_kd',
    #     'wdi_nv_srv_totl_cd',
    #     'wdi_nv_srv_totl_cn',
    #     'wdi_nv_srv_totl_kd',
    #     'wdi_nv_srv_totl_kd_zg',
    #     'wdi_nv_srv_totl_kn',
    #     'wdi_nv_srv_totl_zs',
    #     'wdi_ny_adj_dfor_cd',
    #     'wdi_ny_adj_dmin_gn_zs',
    #     'wdi_ny_adj_dres_gn_zs',
    #     'wdi_ny_adj_ictr_gn_zs',
    #     'wdi_ny_adj_nnty_kd',
    #     'wdi_ny_adj_nnty_kd_zg',
    #     'wdi_ny_gdp_coal_rt_zs',
    #     'wdi_ny_gdp_defl_kd_zg',
    #     'wdi_ny_gdp_defl_kd_zg_ad',
    #     'wdi_ny_gdp_defl_zs',
    #     'wdi_ny_gdp_defl_zs_ad',
    #     'wdi_ny_gdp_disc_cn',
    #     'wdi_ny_gdp_disc_kn',
    #     'wdi_ny_gdp_fcst_cd',
    #     'wdi_ny_gdp_fcst_cn',
    #     'wdi_ny_gdp_fcst_kd',
    #     'wdi_ny_gdp_fcst_kn',
    #     'wdi_ny_gdp_frst_rt_zs',
    #     'wdi_ny_gdp_minr_rt_zs',
    #     'wdi_ny_gdp_mktp_cd',
    #     'wdi_ny_gdp_mktp_cn',
    #     'wdi_ny_gdp_mktp_cn_ad',
    #     'wdi_ny_gdp_mktp_kd',
    #     'wdi_ny_gdp_mktp_kd_zg',
    #     'wdi_ny_gdp_mktp_kn',
    #     'wdi_ny_gdp_mktp_pp_cd',
    #     'wdi_ny_gdp_mktp_pp_kd',
    #     'wdi_ny_gdp_ngas_rt_zs',
    #     'wdi_ny_gdp_pcap_cd',
    #     'wdi_ny_gdp_pcap_cn',
    #     'wdi_ny_gdp_pcap_kd',
    #     'wdi_ny_gdp_pcap_kd_zg',
    #     'wdi_ny_gdp_pcap_kn',
    #     'wdi_ny_gdp_pcap_pp_cd',
    #     'wdi_ny_gdp_pcap_pp_kd',
    #     'wdi_ny_gdp_petr_rt_zs',
    #     'wdi_ny_gdp_totl_rt_zs',
    #     'wdi_ny_gnp_mktp_kd',
    #     'wdi_ny_gnp_mktp_pp_kd',
    #     'wdi_per_si_allsi_cov_pop_tot',
    #     'wdi_per_si_allsi_cov_q1_tot',
    #     'wdi_per_si_allsi_cov_q2_tot',
    #     'wdi_per_si_allsi_cov_q3_tot',
    #     'wdi_per_si_allsi_cov_q4_tot',
    #     'wdi_per_si_allsi_cov_q5_tot',
    #     'wdi_se_adt_1524_lt_fe_zs',
    #     'wdi_se_adt_1524_lt_ma_zs',
    #     'wdi_se_adt_1524_lt_zs',
    #     'wdi_se_adt_litr_fe_zs',
    #     'wdi_se_adt_litr_ma_zs',
    #     'wdi_se_adt_litr_zs',
    #     'wdi_se_enr_prim_fm_zs',
    #     'wdi_se_enr_prsc_fm_zs',
    #     'wdi_se_prm_cmpt_zs',
    #     'wdi_se_prm_cuat_fe_zs',
    #     'wdi_se_prm_cuat_ma_zs',
    #     'wdi_se_prm_cuat_zs',
    #     'wdi_se_prm_enrr',
    #     'wdi_se_prm_nenr',
    #     'wdi_se_prm_tenr_fe',
    #     'wdi_se_prm_tenr_ma',
    #     'wdi_se_sec_cmpt_lo_zs',
    #     'wdi_se_sec_cuat_lo_fe_zs',
    #     'wdi_se_sec_cuat_lo_ma_zs',
    #     'wdi_se_sec_cuat_lo_zs',
    #     'wdi_se_sec_nenr',
    #     'wdi_se_ter_cuat_do_fe_zs',
    #     'wdi_se_ter_cuat_do_ma_zs',
    #     'wdi_se_ter_cuat_do_zs',
    #     'wdi_sg_gen_parl_zs',
    #     'wdi_sg_vaw_reas_zs',
    #     'wdi_sh_dyn_0514',
    #     'wdi_sh_dyn_mort',
    #     'wdi_sh_dyn_mort_fe',
    #     'wdi_sh_dyn_mort_ma',
    #     'wdi_sh_h2o_basw_ru_zs',
    #     'wdi_sh_h2o_basw_ur_zs',
    #     'wdi_sh_h2o_basw_zs',
    #     'wdi_sh_mmr_risk_zs',
    #     'wdi_sh_sta_bass_ru_zs',
    #     'wdi_sh_sta_bass_ur_zs',
    #     'wdi_sh_sta_bass_zs',
    #     'wdi_sh_sta_maln_fe_zs',
    #     'wdi_sh_sta_maln_ma_zs',
    #     'wdi_sh_sta_maln_zs',
    #     'wdi_sh_sta_mmrt',
    #     'wdi_sh_sta_mmrt_ne',
    #     'wdi_sh_sta_stnt_fe_zs',
    #     'wdi_sh_sta_stnt_ma_zs',
    #     'wdi_sh_sta_stnt_zs',
    #     'wdi_sh_sta_traf_p5',
    #     'wdi_sh_sta_wash_p5',
    #     'wdi_sh_svr_wast_fe_zs',
    #     'wdi_sh_svr_wast_ma_zs',
    #     'wdi_sh_svr_wast_zs',
    #     'wdi_si_dst_02nd_20',
    #     'wdi_si_dst_03rd_20',
    #     'wdi_si_dst_04th_20',
    #     'wdi_si_dst_05th_20',
    #     'wdi_si_dst_10th_10',
    #     'wdi_si_dst_frst_10',
    #     'wdi_si_dst_frst_20',
    #     'wdi_si_pov_dday',
    #     'wdi_si_pov_gaps',
    #     'wdi_si_pov_gini',
    #     'wdi_si_pov_lmic',
    #     'wdi_si_pov_umic',
    #     'wdi_sl_agr_empl_ma_zs',
    #     'wdi_sl_agr_empl_zs',
    #     'wdi_sl_ind_empl_zs',
    #     'wdi_sl_srv_empl_zs',
    #     'wdi_sl_tlf_totl_fe_zs',
    #     'wdi_sl_uem_advn_fe_zs',
    #     'wdi_sl_uem_advn_ma_zs',
    #     'wdi_sl_uem_advn_zs',
    #     'wdi_sl_uem_neet_fe_zs',
    #     'wdi_sl_uem_neet_ma_zs',
    #     'wdi_sl_uem_neet_zs',
    #     'wdi_sl_uem_totl_fe_zs',
    #     'wdi_sl_uem_totl_ma_zs',
    #     'wdi_sl_uem_totl_zs',
    #     'wdi_sm_pop_netm',
    #     'wdi_sm_pop_refg',
    #     'wdi_sm_pop_refg_or',
    #     'wdi_sm_pop_totl_zs',
    #     'wdi_sn_itk_defc_zs',
    #     'wdi_sp_dyn_amrt_fe',
    #     'wdi_sp_dyn_amrt_ma',
    #     'wdi_sp_dyn_imrt_fe_in',
    #     'wdi_sp_dyn_imrt_in',
    #     'wdi_sp_dyn_imrt_ma_in',
    #     'wdi_sp_dyn_le00_fe_in',
    #     'wdi_sp_dyn_le00_in',
    #     'wdi_sp_dyn_le00_ma_in',
    #     'wdi_sp_dyn_tfrt_in',
    #     'wdi_sp_dyn_wfrt',
    #     'wdi_sp_hou_fema_zs',
    #     'wdi_sp_pop_0014_fe_zs',
    #     'wdi_sp_pop_0014_ma_zs',
    #     'wdi_sp_pop_0014_to_zs',
    #     'wdi_sp_pop_1564_fe_zs',
    #     'wdi_sp_pop_1564_ma_zs',
    #     'wdi_sp_pop_1564_to_zs',
    #     'wdi_sp_pop_65up_fe_zs',
    #     'wdi_sp_pop_65up_ma_zs',
    #     'wdi_sp_pop_65up_to_zs',
    #     'wdi_sp_pop_dpnd',
    #     'wdi_sp_pop_dpnd_ol',
    #     'wdi_sp_pop_dpnd_yg',
    #     'wdi_sp_pop_grow',
    #     'wdi_sp_pop_totl',
    #     'wdi_sp_rur_totl_zg',
    #     'wdi_sp_rur_totl_zs',
    #     'wdi_sp_urb_grow',
    #     'wdi_sp_urb_totl_in_zs',
    #     'wdi_st_int_arvl',
    #     'wdi_st_int_rcpt_xp_zs',
    #     'wdi_tx_val_agri_zs_un',
    #     'wdi_tx_val_food_zs_un',
    #     'wdi_tx_val_fuel_zs_un',
    #     'wdi_tx_val_mmtl_zs_un',
    #     'wdi_tx_val_tech_mf_zs',
    #     'wdi_vc_idp_nwcv',
    #     'wdi_vc_idp_nwds',
    #     'wdi_vc_idp_tocv',
    #     'wdi_vc_pkp_totl_un'
    # ],
    # Feature collection of CM_DV_ONLY
    # 'features': ['country_name',
    #              'month',
    #              'ged_best_sb',
    #              'ged_count_sb',
    #              'ged_dummy_sb',
    #              'greq_100_ged_best_sb',
    #              'greq_100_splag_1_1_ged_best_sb',
    #              'greq_1_ged_best_sb',
    #              'greq_25_ged_best_sb',
    #              'greq_25_splag_1_1_ged_best_sb',
    #              'greq_500_ged_best_sb',
    #              'greq_500_splag_1_1_ged_best_sb',
    #              'greq_5_ged_best_sb',
    #              'greq_5_splag_1_1_ged_best_sb',
    #              'ln_ged_best_sb',
    #              'onset_24_greq_100_ged_best_sb',
    #              'onset_24_greq_1_ged_best_sb',
    #              'onset_24_greq_25_ged_best_sb',
    #              'onset_24_greq_500_ged_best_sb',
    #              'onset_24_greq_5_ged_best_sb',
    #              'rollmax_24_ged_dummy_sb',
    #              'rollmax_24_greq_5_ged_best_sb',
    #              'splag_1_1_ged_best_sb',
    #              'splag_1_1_ged_dummy_sb',
    #              'time_since_ged_dummy_sb',
    #              'time_since_greq_100_ged_best_sb',
    #              'time_since_greq_100_splag_1_1_ged_best_sb',
    #              'time_since_greq_25_ged_best_sb',
    #              'time_since_greq_25_splag_1_1_ged_best_sb',
    #              'time_since_greq_500_ged_best_sb',
    #              'time_since_greq_500_splag_1_1_ged_best_sb',
    #              'time_since_greq_5_ged_best_sb',
    #              'time_since_greq_5_splag_1_1_ged_best_sb',
    #              'time_since_splag_1_1_ged_dummy_sb',
    #              'tlag_10_greq_100_ged_best_sb',
    #              'tlag_10_greq_1_ged_best_sb',
    #              'tlag_10_greq_25_ged_best_sb',
    #              'tlag_10_greq_500_ged_best_sb',
    #              'tlag_10_greq_5_ged_best_sb',
    #              'tlag_11_greq_100_ged_best_sb',
    #              'tlag_11_greq_1_ged_best_sb',
    #              'tlag_11_greq_25_ged_best_sb',
    #              'tlag_11_greq_500_ged_best_sb',
    #              'tlag_11_greq_5_ged_best_sb',
    #              'tlag_12_greq_100_ged_best_sb',
    #              'tlag_12_greq_1_ged_best_sb',
    #              'tlag_12_greq_25_ged_best_sb',
    #              'tlag_12_greq_500_ged_best_sb',
    #              'tlag_12_greq_5_ged_best_sb',
    #              'tlag_1_greq_100_ged_best_sb',
    #              'tlag_1_greq_1_ged_best_sb',
    #              'tlag_1_greq_25_ged_best_sb',
    #              'tlag_1_greq_500_ged_best_sb',
    #              'tlag_1_greq_5_ged_best_sb',
    #              'tlag_1_splag_1_1_ged_best_sb',
    #              'tlag_2_greq_100_ged_best_sb',
    #              'tlag_2_greq_1_ged_best_sb',
    #              'tlag_2_greq_25_ged_best_sb',
    #              'tlag_2_greq_500_ged_best_sb',
    #              'tlag_2_greq_5_ged_best_sb',
    #              'tlag_3_greq_100_ged_best_sb',
    #              'tlag_3_greq_1_ged_best_sb',
    #              'tlag_3_greq_25_ged_best_sb',
    #              'tlag_3_greq_500_ged_best_sb',
    #              'tlag_3_greq_5_ged_best_sb',
    #              'tlag_4_greq_100_ged_best_sb',
    #              'tlag_4_greq_1_ged_best_sb',
    #              'tlag_4_greq_25_ged_best_sb',
    #              'tlag_4_greq_500_ged_best_sb',
    #              'tlag_4_greq_5_ged_best_sb',
    #              'tlag_5_greq_100_ged_best_sb',
    #              'tlag_5_greq_1_ged_best_sb',
    #              'tlag_5_greq_25_ged_best_sb',
    #              'tlag_5_greq_500_ged_best_sb',
    #              'tlag_5_greq_5_ged_best_sb',
    #              'tlag_6_greq_100_ged_best_sb',
    #              'tlag_6_greq_1_ged_best_sb',
    #              'tlag_6_greq_25_ged_best_sb',
    #              'tlag_6_greq_500_ged_best_sb',
    #              'tlag_6_greq_5_ged_best_sb',
    #              'tlag_7_greq_100_ged_best_sb',
    #              'tlag_7_greq_1_ged_best_sb',
    #              'tlag_7_greq_25_ged_best_sb',
    #              'tlag_7_greq_500_ged_best_sb',
    #              'tlag_7_greq_5_ged_best_sb',
    #              'tlag_8_greq_100_ged_best_sb',
    #              'tlag_8_greq_1_ged_best_sb',
    #              'tlag_8_greq_25_ged_best_sb',
    #              'tlag_8_greq_500_ged_best_sb',
    #              'tlag_8_greq_5_ged_best_sb',
    #              'tlag_9_greq_100_ged_best_sb',
    #              'tlag_9_greq_1_ged_best_sb',
    #              'tlag_9_greq_25_ged_best_sb',
    #              'tlag_9_greq_500_ged_best_sb',
    #              'tlag_9_greq_5_ged_best_sb',
    #              ],
    'labels': ['dv'],  # Fixed, don't change
    # Adjust TRAIN\VAL\TEST split below
    'train_split': [121, 396],
    'val_split': [397, 432],
    'test_split': [433, 468],

    'd3m_root': '/home/yulin/Desktop/final_datasets/',  # Any path you want to store the generated dataset
    'data_name': 'cm_s6',  # Name of the generated dataset
    'mode': 'train_val'  # Just use this value for simplicity.
}

with open('../test_config.json', 'w') as outfile:
    json.dump(template, outfile)
